import argparse
import os
import json
import subprocess
from logger import ULogger
import rknn_quant
from model_repo_monitor import ModelRepoMonitor
import multiprocessing
def parse_args():
    args_parser=argparse.ArgumentParser()
    args_parser.add_argument("--repo_dir",type=str,required=True)
    args=args_parser.parse_args()
    return args

def rknn_quant_proxy(config):
    return subprocess.call("python rknn_quant.py {0}".format(config),shell=True)

if __name__=="__main__":
    args = parse_args()
    model_repo_monitor = ModelRepoMonitor(args.repo_dir)
    model_repo_monitor.register_rknn_quant_func(rknn_quant.run_quant)

    model_repo_monitor.start_monitor()


